package com.milvus.config;

import io.milvus.v2.client.ConnectConfig;
import io.milvus.v2.client.MilvusClientV2;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import java.io.Serializable;

@Configuration
public class MilvusClientV2Config implements Serializable {

    @Autowired
    MilvusConfig milvusConfig;
    @Bean(name = "milvusClient")
    public MilvusClientV2 milvusClientV2()
    {
        return new MilvusClientV2(
                ConnectConfig.builder()
                        .uri(milvusConfig.uri)
                        .token(milvusConfig.token)
                        .build()
        );
    }
}
